package com.frank.leetcode.easy;

import org.junit.Test;

import java.util.Arrays;
import java.util.Iterator;
import java.util.LinkedHashSet;
import java.util.Set;

/**
 * https://leetcode-cn.com/explore/interview/card/top-interview-questions-easy/1/array/21/
 * 从排序数组中删除重复项
 * <p>
 * 给定一个排序数组，你需要在原地删除重复出现的元素，使得每个元素只出现一次，返回移除后数组的新长度。
 * 不要使用额外的数组空间，你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。
 * <p>
 * 示例 1:
 * 给定数组 nums = [1,1,2],
 * 函数应该返回新的长度 2, 并且原数组 nums 的前两个元素被修改为 1, 2。
 * 你不需要考虑数组中超出新长度后面的元素。
 * <p>
 * 示例 2:
 * 给定 nums = [0,0,1,1,1,2,2,3,3,4],
 * 函数应该返回新的长度 5, 并且原数组 nums 的前五个元素被修改为 0, 1, 2, 3, 4。
 * 你不需要考虑数组中超出新长度后面的元素。
 * <p>
 * Created by zhy on 2018/6/21.
 */
public class RemoveDuplicates {

    @Test
    public void run() {
        int[] nums = new int[]{0, 0, 1, 1, 1, 2, 2, 3, 3, 4};

        int val = _removeDuplicates(nums);
        System.out.println("位数: " + val);
        System.out.println(Arrays.toString(nums));
    }

    private int removeDuplicates(int[] nums) {
        if (nums.length < 2) {
            return nums.length;
        }
        Set set = new LinkedHashSet();

        for (int i = 0; i < nums.length; i++) {
            set.add(nums[i]);
        }

        int[] temp = new int[set.size()];
        Iterator iterator = set.iterator();
        int index = 0;
        while (iterator.hasNext()) {
            temp[index] = (int) iterator.next();
            index++;
        }
        System.arraycopy(temp, 0, nums, 0, temp.length);
        return temp.length;
    }

    /**
     * 优化
     */
    private int _removeDuplicates(int[] nums) {
        int num = 0;

        for (int i = 1; i < nums.length; i++) {
            if (nums[num] != nums[i]) {
                num++;
                nums[num] = nums[i];
            }
        }

        return num + 1;
    }
}
